package com.yonyou.train.algorithem;

import javax.crypto.SecretKeyFactory;
import javax.crypto.spec.PBEKeySpec;
import java.math.BigInteger;
import java.security.NoSuchAlgorithmException;
import java.security.SecureRandom;
import java.security.spec.InvalidKeySpecException;
import java.security.spec.KeySpec;
import java.util.Arrays;

/**
 * 通过哈希算法进行加密。因为哈希算法是单向的，可以将任何大小的数据转换为定长的"指纹"，而且无法被反向计算。另外，即使数据源只改动了一点点，哈希的结果
 * 也会完全不同。这样的特性使得它非常适合用于保存密码，因为我们需要加密后的密码无法被破解，同时也能保证正确校验每个用户的密码。但是哈希加密可以通过字典攻击
 * 和暴力攻击破解。
 * <p>
 * 密码加盐。盐是一个添加到用户的密码哈希过程中的一段随机序列。这个机制能够防止通过预先计算结果的彩虹表破解。每个用户都有自己的盐，这样的结果就是即使用户的密码相同，通过加
 * 盐后哈希值也将不同。为了校验密码是否正确，我们需要储存盐值。通常和密码哈希值一起存放在账户数据库中，或者直接存为哈希字符串的一部分。
 */
public class PBKDF2 {
    public static final String PBKDF2_ALGRITHM = "PBKDF2WithHmacSHA1";
    /**
     * 盐的长度
     */
    public static final int SALT_BYTE_SIZE = 32 / 2;
    /**
     * 生成密文的长度
     */
    public static final int HASH_BIT_SIZE = 128 * 4;
    /**
     * 迭代次数
     */
    public static final int PBKDF2_ITERATIONS = 1000;

    /**
     * 对输入密码进行验证
     */
    public static boolean authenticate(String attemptedPassword, String encryptedPassword, String salt) {
//用相同的盐值对用户输入的密码进行加密
        String encryptedAttemptedPassword = getEncryptedPassword(attemptedPassword, salt);

        return encryptedAttemptedPassword.equals(encryptedPassword);
    }

    /**
     * 生成密文
     */
    public static String getEncryptedPassword(String password, String salt) {
        KeySpec spec = new PBEKeySpec(password.toCharArray(), fromHex(salt), PBKDF2_ITERATIONS, HASH_BIT_SIZE);//传入的四个参数分别是：原密码的字符数组，所加盐的字节数组，

        //迭代次数，生成密文的长度
        try {
            SecretKeyFactory f = SecretKeyFactory.getInstance(PBKDF2_ALGRITHM);

            return toHex(f.generateSecret(spec).getEncoded());//返回十六进制密文
        } catch (NoSuchAlgorithmException e) {
            e.printStackTrace();
        } catch (InvalidKeySpecException e) {
            e.printStackTrace();
        }

        return null;
    }

    /**
     * 将随机盐和密文拼成一起
     */
    public static String getPwd(String password) {
        String salt = generateSalt();
        String encrypted = getEncryptedPassword(password, salt);
        return encrypted + salt;
    }

    /**
     * 通过提供加密的强随机数生成器 生成盐
     */
    public static String generateSalt() {
        try {
            SecureRandom random = SecureRandom.getInstance("SHA1PRNG");
            byte[] salt = new byte[SALT_BYTE_SIZE];
            random.nextBytes(salt);


            return toHex(salt);
        } catch (NoSuchAlgorithmException e) {
            e.printStackTrace();
        }
        return null;
    }

    /**
     * 十六进制字符串转字节数组
     */
    private static byte[] fromHex(String hex) {
        byte[] binary = new byte[hex.length() / 2];
        for (int i = 0; i < binary.length; i++) {
            binary[i] = (byte) Integer.parseInt(hex.substring(2 * i, 2 * i + 2), 16);
        }
        return binary;
    }

    /**
     * 字节数组转十六进制字符串
     */
    private static String toHex(byte[] array) {
        BigInteger bi = new BigInteger(1, array);
        String hex = bi.toString(16);
        int paddingLength = (array.length * 2) - hex.length();
        if (paddingLength > 0)
            return String.format("%0" + paddingLength + "d", 0) + hex;
        else
            return hex;
    }

    /**
     * 测试
     */
    public static void main(String[] args) {
        String s = "263c47dae369bd1741b6dc0a43c4cd7b2731dee08762c3b1cbbef0b087aa5cbb902fec7b20d238c75578be9a12c9101822ca54d7969db8dfa8e83482f1cff606d82ec2d300d8a02c60c3489f08cfe1c8";
        System.out.println(s.length());

    }
}
















































